package dao

import (
	"database/sql"
	"fmt"
	"news/setting"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

// DB 数据库对象
var DB *gorm.DB
var db *sql.DB

// InitMySQL 初始化数据库
func InitMySQL(cfg *setting.MySQLConfig) (err error) {
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		cfg.User, cfg.Password, cfg.Host, cfg.Port, cfg.DB)
	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
		QueryFields: true,
	})
	if err != nil {
		return
	}
	db, err = DB.DB()
	return db.Ping()
}

// Close 关闭数据库
func Close() {
	db.Close()
}
